#pragma once

#include <iostream>
#include <functional>
#include <string>
#include <cassert>

#include <signal.h>
#include "log.hpp"
#include "Sock.hpp"

class HttpServer
{
public:
    using func_t = std::function<void(int)>;

    HttpServer(uint16_t port, func_t func)
        :_port(port)
        ,_func(func)
    {
        _listensock = sock.Socket();
        sock.Bind(_listensock, _port);
        sock.Listen(_listensock);
    }

    void start()
    {
        signal(SIGCHLD, SIG_IGN);
        while (true)
        {
            uint16_t client_port;
            std::string client_ip;
            int servicesock = sock.Accpet(_listensock, &client_port, &client_ip);
            if (servicesock < 0)    continue;

            // 使用子进程来完成任务
            pid_t id = fork();
            assert(id >= 0);
            if (id == 0)
            {
                // 子进程
                close(_listensock);
                _func(servicesock);
                close(servicesock);
                exit(0);
            }
            close(servicesock);
        }
    }

    ~HttpServer()
    {
        if (_listensock >= 0) close(_listensock);
    }
private:
    int _listensock;
    Sock sock;
    uint16_t _port;
    func_t _func;
};
